package Mode;

import java.util.concurrent.PriorityBlockingQueue;

/**
 * 创建一个自己的定时器
 *
 * 步骤:
 *      1) 描述一个任务 : runnable + time
 *      2) 用堆来组织任务
 *      3) 实现shchedule 来让任务进入堆
 *      4) 创建扫描线程获取队首元素, 判断时间是否到达
 *
 *
 * 要注意两个问题:
 *      1) MyTask的时间比较规则要自己设定, 因为创建了堆
 *      2) 用到了while (true)  会一直循环, 比如任务时间还没到的时候, 就不停地把任务拿出来,再放回去
 * 这样的操作叫做忙等, 我们可以用wait来解决.
 *      这里用wait, 为啥不用sleep呢? 因为sleep不能中途唤醒, 如果在任务之间又插入一条需要先执行的任务
 * 此时就需要先去完成这个任务, 需要中途唤醒
 */

//创建一个类, 表示任务(这个任务是放在堆里的, 需要自己手动设置比较规则)
class MyTask implements Comparable<MyTask>{
    //任务工作内容
    private Runnable runnable;
    //任务具体时间干啥,保存时间戳
    private long time;

    //安排任务一般用如1小时后干啥啥,而不用具体时间,所以这里用after来表示多少时间后干啥
    //注意一下, 类里的time是具体的执行时间,我们只是为了方便安排任务,所以设置了一个after来表示after时间后干什么
    public MyTask(Runnable runnable, long after) {
        this.runnable = runnable;
        this.time = System.currentTimeMillis() + after;
    }

    public void run(){
        runnable.run();
    }

    public long getTime(){
        return time;
    }

    @Override
    public int compareTo(MyTask o) {
        return (int) (this.time - o.time);
    }
}

class MyTimer{
    /*
        定时器来组织任务,由于需要快速找到时间最小的那个任务,所以数据结构就用堆
        但是此处的堆还要考虑线程安全问题,所以用阻塞队列
        但是创建了堆,别忘了实现比较方法,毕竟自定义类需要自己去规定比较的规则
    */
    private PriorityBlockingQueue<MyTask> queue = new PriorityBlockingQueue();
    private Object locker = new Object();

    //创建一个schedule方法把计划的任务都放进去
    public void schedule(Runnable runnable, long after){
        MyTask myTask = new MyTask(runnable, after);
        queue.offer(myTask);
        //放了一个任务就要唤醒一次,看看是否需要完成
        synchronized (locker){
            locker.notify();
        }
    }

    //构造方法中用线程来找哪个任务时间到了需要执行
    public MyTimer(){
        Thread thread = new Thread(() -> {
            while (true){
                try {
                    MyTask task = queue.take();
                    long curTime = System.currentTimeMillis();
                    if (curTime < task.getTime()){
                        //时间还没到,放回去
                        queue.offer(task);
                        //放回去就等到了时间再拿
                        synchronized (locker){
                            locker.wait(task.getTime() - curTime);
                        }
                    }else {
                        //到了就运行
                        task.run();
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
        thread.start();
    }
}

public class Test4 {
    public static void main(String[] args) {
        MyTimer myTimer = new MyTimer();
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("thread");
            }
        }, 3000);
        System.out.println("main");
    }
}